import torch

Adj_PA = torch.FloatTensor([
    [1, 0, 1],
    [1, 0, 0],
    [0, 1, 1],
    [0, 1, 0],
    [0, 1, 0],
])
Adj_AP = Adj_PA.t()
print(Adj_PA)
print(Adj_AP)
print(Adj_PA @ Adj_AP)

conv1 = torch.nn.Conv1d(5,2,kernel_size=1)
print(conv1(Adj_PA))


conv2 = torch.nn.Conv1d(3,2,kernel_size=1)
print(conv2(Adj_AP))